Return statement in Python